package com.kt.emall.order.config;

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

/**
 * @program: e-mall
 * @description: Rabbit消息类型配置
 * @author: KamTang
 * @create: 2022-01-19 13:49
 **/
@Configuration
public class OrderRabbitConfig {

    @Autowired
    RabbitTemplate rabbitTemplate;

    /**
     * 将对象以json形式传输
     * @return
     */
    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制RabbitTemplate
     * 1、服务器收到消息就回调
     *      1)、@PostConstruct：OrderRabbitConfig对象创建完成之后，再执行这个方法
     *      2)、设置确认回调
     * 2、消息正确抵达队列进行回调
     *      1)、spring.rabbit.publisher-returns: true
     *          spring.rabbit.template.mandatory: true
     *      2)、设置确认回调ReturnCallback
     * 3、消费端确认（保证每个消息被正确“消费”，此时才可以被broker删除这个消息）
     * 注：一定要加上：spring.rabbitmq.listener.simple.acknowledge-mode=manual  手动接收
     *      1、默认是自动确认，只要消息接收到，客户端会自动确认，服务端就会移除这个消息
     *          问题：收到很多消息，自动回复给服务器ack，但只有一个消息处理成功，发生宕机。消息流失
     *               消费者手动确认。只要我们没有明确告诉MQ消息已经被接收，没有ack。那么在队列中的消息一直会是Unacked状态
     *               即使客户端宕机，消息也不会丢失，会变成Ready状态，下一次有新的客户端连接进来就发给它。
     *      2、如何接收
     *          channel.basicAck(deliveryTag， false);接收——业务成功完成就应该接收
     *          channel.basicNack(deliveryTag， false, true);拒绝——业务失败
     */
    @PostConstruct
    public void initRabbitTemplate() {
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * 1、只要消息抵达Broker，ack=true
             * @param correlationData：当前消息的唯一关联数据（这个是消息的唯一ID）
             * @param ack：消息是否成功收到
             * @param cause：失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                // do something
            }
        });

        // 设置消息抵达队列的确认回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只要消息没有传递给指定的队列，就触发这个失败回调
             * @param message：投递失败的消息详细信息
             * @param replyCode：回复的状态码
             * @param replyText：回复的文本内容
             * @param exchange：消息发送的目的交换机
             * @param routingKey：消息使用的路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                // do something
            }
        });
    }
}
